import pytest
import requests

BASE_URL = "https://api.example.com"

@pytest.mark.smoke
class TestSmokeTests:
    """冒烟测试套件"""
    
    def test_server_is_running(self):
        """测试服务器是否运行"""
        response = requests.get(f"{BASE_URL}/health")
        assert response.status_code == 200
    
    def test_database_connection(self):
        """测试数据库连接"""
        response = requests.get(f"{BASE_URL}/health/db")
        assert response.status_code == 200
        assert response.json()["status"] == "connected"
    
    def test_user_can_login(self):
        """测试用户能否登录"""
        response = requests.post(f"{BASE_URL}/login", json={
            "email": "test@example.com",
            "password": "password123"
        })
        assert response.status_code == 200
        assert "token" in response.json()
    
    def test_user_can_search_products(self):
        """测试用户能否搜索商品"""
        response = requests.get(f"{BASE_URL}/products/search?q=iPhone")
        assert response.status_code == 200
        assert "products" in response.json()
    
    def test_user_can_add_to_cart(self):
        """测试用户能否加入购物车"""
        # 先登录
        login_response = requests.post(f"{BASE_URL}/login", json={
            "email": "test@example.com",
            "password": "password123"
        })
        token = login_response.json()["token"]
        
        # 加入购物车
        response = requests.post(
            f"{BASE_URL}/cart/add",
            json={"product_id": 123, "quantity": 1},
            headers={"Authorization": f"Bearer {token}"}
        )
        assert response.status_code == 200

# 运行冒烟测试：
# pytest -m smoke -v
# 耗时：15-30 秒

# 如果冒烟测试通过，再运行完整测试：
# pytest -v
# 耗时：2-3 小时

# 冒烟测试发现问题示例
def test_critical_api_endpoints():
    """测试关键 API 端点"""
    critical_endpoints = [
        "/health",
        "/login",
        "/products",
        "/cart",
        "/orders"
    ]
    
    for endpoint in critical_endpoints:
        response = requests.get(f"{BASE_URL}{endpoint}")
        
        # 如果任何关键端点失败，立即报告
        assert response.status_code in [200, 401], \
            f"关键端点 {endpoint} 失败！状态码：{response.status_code}"

# 结果：
# ❌ 关键端点 /cart 失败！状态码：500
# → 立即退回开发，不继续测试

# CI/CD 中的冒烟测试
# .github/workflows/smoke-test.yml
name: Smoke Tests

on:
  push:
    branches: [main, develop]
  pull_request:
    branches: [main]

jobs:
  smoke-test:
    runs-on: ubuntu-latest
    
    steps:
      - uses: actions/checkout@v2
      
      - name: Set up Python
        uses: actions/setup-python@v2
        with:
          python-version: 3.9
      
      - name: Install dependencies
        run: pip install -r requirements.txt
      
      - name: Run smoke tests
        run: pytest -m smoke -v --tb=short
      
      - name: Notify on failure
        if: failure()
        run: |
          echo "❌ 冒烟测试失败！请立即修复。"
          exit 1
      
      - name: Continue to full tests
        if: success()
        run: pytest -v

# 冒烟测试清单
smoke_test_checklist = [
    "服务器能否启动",
    "数据库能否连接",
    "用户能否登录",
    "能否搜索商品",
    "能否加入购物车",
    "能否下单",
    "能否支付"
]

@pytest.mark.smoke
def test_complete_user_flow():
    """测试完整用户流程（冒烟测试）"""
    # 1. 登录
    login_response = requests.post(f"{BASE_URL}/login", json={
        "email": "test@example.com",
        "password": "password123"
    })
    assert login_response.status_code == 200
    token = login_response.json()["token"]
    
    # 2. 搜索商品
    search_response = requests.get(
        f"{BASE_URL}/products/search?q=iPhone",
        headers={"Authorization": f"Bearer {token}"}
    )
    assert search_response.status_code == 200
    
    # 3. 加入购物车
    cart_response = requests.post(
        f"{BASE_URL}/cart/add",
        json={"product_id": 123, "quantity": 1},
        headers={"Authorization": f"Bearer {token}"}
    )
    assert cart_response.status_code == 200
    
    # 4. 下单
    order_response = requests.post(
        f"{BASE_URL}/orders",
        json={"address": "北京市朝阳区xxx"},
        headers={"Authorization": f"Bearer {token}"}
    )
    assert order_response.status_code == 201

# 如果冒烟测试通过
if smoke_tests_passed:
    print("✅ 冒烟测试通过，继续完整测试")
    run_full_tests()
else:
    print("❌ 冒烟测试失败，退回开发")
    notify_developers()
    reject_build()

# ✅ 好的冒烟测试：简单、快速
@pytest.mark.smoke
def test_api_is_accessible():
    response = requests.get(f"{BASE_URL}/health")
    assert response.status_code == 200

# ❌ 不好的冒烟测试：复杂、慢
@pytest.mark.smoke
def test_complex_business_logic():
    # 100+ 行复杂逻辑
    pass

# 核心功能清单
core_features = [
    "用户认证",
    "数据库连接",
    "核心 API",
    "关键业务流程"
]

# 为每个核心功能编写冒烟测试
@pytest.mark.smoke
def test_user_authentication():
    """测试用户认证"""
    pass

@pytest.mark.smoke
def test_database_connection():
    """测试数据库连接"""
    pass